home *** CD-ROM | disk | FTP | other *** search
/ Mac Easy 2010 May / Mac Life Ubuntu.iso / casper / filesystem.squashfs / usr / lib / python2.6 / lib2to3 / fixes / fix_execfile.py < prev    next >
Encoding:
Python Source  |  2009-04-18  |  1.9 KB  |  52 lines
  1. # Copyright 2006 Google, Inc. All Rights Reserved.
  2. # Licensed to PSF under a Contributor Agreement.
  3.  
  4. """Fixer for execfile.
  5.  
  6. This converts usages of the execfile function into calls to the built-in
  7. exec() function.
  8. """
  9.  
  10. from .. import fixer_base
  11. from ..fixer_util import (Comma, Name, Call, LParen, RParen, Dot, Node,
  12.                           ArgList, String, syms)
  13.  
  14.  
  15. class FixExecfile(fixer_base.BaseFix):
  16.  
  17.     PATTERN = """
  18.     power< 'execfile' trailer< '(' arglist< filename=any [',' globals=any [',' locals=any ] ] > ')' > >
  19.     |
  20.     power< 'execfile' trailer< '(' filename=any ')' > >
  21.     """
  22.  
  23.     def transform(self, node, results):
  24.         assert results
  25.         filename = results["filename"]
  26.         globals = results.get("globals")
  27.         locals = results.get("locals")
  28.  
  29.         # Copy over the prefix from the right parentheses end of the execfile
  30.         # call.
  31.         execfile_paren = node.children[-1].children[-1].clone()
  32.         # Construct open().read().
  33.         open_args = ArgList([filename.clone()], rparen=execfile_paren)
  34.         open_call = Node(syms.power, [Name("open"), open_args])
  35.         read = [Node(syms.trailer, [Dot(), Name('read')]),
  36.                 Node(syms.trailer, [LParen(), RParen()])]
  37.         open_expr = [open_call] + read
  38.         # Wrap the open call in a compile call. This is so the filename will be
  39.         # preserved in the execed code.
  40.         filename_arg = filename.clone()
  41.         filename_arg.set_prefix(" ")
  42.         exec_str = String("'exec'", " ")
  43.         compile_args = open_expr + [Comma(), filename_arg, Comma(), exec_str]
  44.         compile_call = Call(Name("compile"), compile_args, "")
  45.         # Finally, replace the execfile call with an exec call.
  46.         args = [compile_call]
  47.         if globals is not None:
  48.             args.extend([Comma(), globals.clone()])
  49.         if locals is not None:
  50.             args.extend([Comma(), locals.clone()])
  51.         return Call(Name("exec"), args, prefix=node.get_prefix())
  52.